#include "ForceGenerator.hpp"

class ForceGeneratorSpring : public ForceGenerator
{
protected:
	using ForceGenerator::entityRegister;
	map<int, Entity*> *springRegister;

	float springConstant;
	float springDistance;
public:
	ForceGeneratorSpring(float constant, float distance);
	~ForceGeneratorSpring();

	void Register(int id, Entity *source, Entity *other);
	void Update(int id, float time);
};